#pragma once
#include <pthread.h>
#include <vector>
#include <queue>
const int defaultcap = 5;


namespace kzz
{

template <class T>
class BlockQueue
{
private:
    bool IsFull()
    {
        return _q.size() >= _cap;
    }
    bool Empty()
    {
        return _q.empty();
    }

public:
    BlockQueue(int cap = defaultcap)
        : _cap(cap), _csleep_num(0), _psleep_num(0)
    {
        pthread_mutex_init(&_mutex, nullptr);
        pthread_cond_init(&_full_cond, nullptr);
        pthread_cond_init(&_empty_cond, nullptr);
    }

    // 生产者使用
    void Equeue(const T &in)
    {
        // 必须先加锁，判断也需要访问临界资源
        pthread_mutex_lock(&_mutex);

        // if(IsFUll())    //错误
        while (IsFull())
        {
            // 如果临界资源满了，让线程去等待
            _psleep_num++;
            pthread_cond_wait(&_full_cond, &_mutex);
            // 等待完毕，现在已经重新获得锁了，但是现在是否为满并不清楚，所以判断条件一定要用while
            _psleep_num--;
        }
        // 完成生产者的写入
        _q.push(in);
        // 现在临界资源里是一定有资源的，通知消费者来消费
        if(_csleep_num>0)
        pthread_cond_signal(&_empty_cond);
        // 完成生产过程，解锁
        pthread_mutex_unlock(&_mutex);
    }

    // 消费者使用，拿出临界资源的数据
    T Pop()
    {
        pthread_mutex_lock(&_mutex);
        while (Empty())
        {
            // 如果临界资源为空，消费者线程去等待
            _csleep_num++;
            pthread_cond_wait(&_empty_cond,&_mutex);
            // 等待结束
            _csleep_num--;
        }
        //取数据
        T data=_q.front();
        _q.pop();
      
        // 当前临界资源一定有空位置
        if(_psleep_num>0)
        pthread_cond_signal(&_full_cond);
        
        pthread_mutex_unlock(&_mutex);
        return data;
    }

    ~BlockQueue()
    {
        pthread_mutex_destroy(&_mutex);
        pthread_cond_destroy(&_full_cond);
        pthread_cond_destroy(&_empty_cond);
    }

private:
    std::queue<T> _q; // 临界资源
    int _cap;

    pthread_mutex_t _mutex;
    pthread_cond_t _full_cond;
    pthread_cond_t _empty_cond;

    int _csleep_num; //
    int _psleep_num;
};
}